package com.spzx.common.rabbit.service;

import com.alibaba.fastjson2.JSON;
import com.spzx.common.core.utils.uuid.UUID;
import com.spzx.common.rabbit.entity.GuiguCorrelationData;
import org.springframework.amqp.AmqpException;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.core.MessagePostProcessor;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;

import java.util.concurrent.TimeUnit;

@Service
public class RabbitService {

    @Autowired
    private RabbitTemplate rabbitTemplate;

    @Autowired
    private RedisTemplate redisTemplate;

    /**
     * 发送普通消息
     *
     * @param exchange   交换机
     * @param routingKey 路由键
     * @param message    消息
     */
/*    public boolean sendMessage(String exchange, String routingKey, Object message) {
        rabbitTemplate.convertAndSend(exchange, routingKey, message);
        return true;
    }*/

    //发送消息时需要携带关联数据，当消息发送失败时，可以通过回退的关联数据进行消息重发
    public boolean sendMessage(String exchange, String routingKey, Object message) {
        //声明关联数据: 类似邮件 的 附件
        String id = "mq:" + UUID.randomUUID().toString().replaceAll("-", "");
        GuiguCorrelationData guiguCorrelationData = new GuiguCorrelationData();
        guiguCorrelationData.setId(id);
        guiguCorrelationData.setMessage(message);
        guiguCorrelationData.setExchange(exchange);
        guiguCorrelationData.setRoutingKey(routingKey);

        //将关联数据缓存到redis中：   原因：回退回调函数只能接收回调的关联对象的id。需要根据id从缓存中找回关联数据，进行消息重发。
        //失效时间应该大于所有重试需要时间和。
        redisTemplate.opsForValue().set(id, JSON.toJSONString(guiguCorrelationData),10, TimeUnit.MINUTES);

        //必须先保存缓存再发送消息。否则，可能出现消息发送失败需要重试时获取不到缓存数据的情况。
        rabbitTemplate.convertAndSend(exchange, routingKey, message, guiguCorrelationData);

        return true;
    }

    /**
     * 用于发送延迟消息
     * @param exchangeDelay
     * @param routingDelay
     * @param message
     * @param delayTime 单位秒
     */
    public boolean sendDelayMessage(String exchangeDelay, String routingDelay, String message, int delayTime) {
        //声明关联数据: 类似邮件 的 附件
        String id = "mq:" + UUID.randomUUID().toString().replaceAll("-", "");
        GuiguCorrelationData guiguCorrelationData = new GuiguCorrelationData();
        guiguCorrelationData.setId(id);
        guiguCorrelationData.setMessage(message);
        guiguCorrelationData.setExchange(exchangeDelay);
        guiguCorrelationData.setRoutingKey(routingDelay);
        guiguCorrelationData.setDelay(true);
        guiguCorrelationData.setDelayTime(delayTime);

        //将关联数据缓存到redis中：   原因：回退回调函数只能接收回调的关联对象的id。需要根据id从缓存中找回关联数据，进行消息重发。
        //失效时间应该大于所有重试需要时间和。
        redisTemplate.opsForValue().set(id, JSON.toJSONString(guiguCorrelationData),10, TimeUnit.MINUTES);

        //必须先保存缓存再发送消息。否则，可能出现消息发送失败需要重试时获取不到缓存数据的情况。
        rabbitTemplate.convertAndSend(exchangeDelay, routingDelay, message, new MessagePostProcessor() {
            @Override
            public Message postProcessMessage(Message message) throws AmqpException {
                message.getMessageProperties().setDelay(delayTime*1000); //MQ服务器时间单位毫秒。
                return message;
            }
        }, guiguCorrelationData);

        return true;
    }
}